@echo off
REM 检查是否传入参数，如果没有则使用默认值

REM 设置训练过程最大 epochs，防止过拟合
IF NOT "%epoch_time%"=="" (
    SET EPOCH_TIME=%epoch_time%
) ELSE (
    SET EPOCH_TIME=150
)

REM 设置已经训练好的模型加载路径
IF NOT "%out_dir%"=="" (
    SET OUT_DIR=%out_dir%
) ELSE (
    SET OUT_DIR=model_out\Synapse\SwinUnet0
)

REM 项目的配置文件
IF NOT "%cfg%"=="" (
    SET CFG=%cfg%
) ELSE (
    SET CFG=configs\swin_tiny_patch4_window7_224_lite.yaml
)

REM 指定数据的根路径，通常用于验证集或测试集
IF NOT "%data_dir%"=="" (
    SET DATA_DIR=%data_dir%
) ELSE (
    SET DATA_DIR=datasets\Synapse
)

REM 基础学习率
IF NOT "%learning_rate%"=="" (
    SET LEARNING_RATE=%learning_rate%
) ELSE (
    SET LEARNING_RATE=0.05
)

REM 图片大小
IF NOT "%img_size%"=="" (
    SET IMG_SIZE=%img_size%
) ELSE (
    SET IMG_SIZE=224
)

REM 图片批号
IF NOT "%batch_size%"=="" (
    SET BATCH_SIZE=%batch_size%
) ELSE (
    SET BATCH_SIZE=24
)

REM 打印信息
echo start test model

REM 运行 Python 脚本
python test.py --cfg %CFG% --root_path %DATA_DIR% --max_epochs %EPOCH_TIME% --output_dir %OUT_DIR% --img_size %IMG_SIZE% --base_lr %LEARNING_RATE% --batch_size %BATCH_SIZE%
